# from gpt2_vpp_test import pretrain

# new change for test for overlap test 1
#from gpt2_vpp_test_overlap import pretrain

from gpt import pretrain

if __name__ == "__main__":

    pretrain()
